# -*- coding: utf-8 -*-

from PIL import Image
import argparse

parse = argparse.ArgumentParser() #读取命令行参数

parse.add_argument('file') #图片路径
parse.add_argument('-o', '--output') #输出文件
parse.add_argument('-width', type=int, default=80) #输出图片宽度
parse.add_argument('-height', type=int, default=80) #输出图片高度

ascii_char = list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. ") #替换色块的符号

args = parse.parse_args()

IMG = args.file
WIDTH = args.width
HEIGHT = args.height
OUTPUT = args.output


def get_char(r, g, b, aplha=256):
    if aplha == 0:
        return ' '
    length = len(ascii_char)
    gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)

    uint = (256.0 + 1) / length
    return ascii_char[int(gray / uint)]


if __name__ == '__main__':
    im = Image.open(IMG)
    im = im.resize((WIDTH, HEIGHT), Image.NEAREST)
    txt = ""

    for i in range(HEIGHT):
        for j in range(WIDTH):
            txt += get_char(*im.getpixel((j, i)))
        txt += '\n'

    print(txt)

    if OUTPUT:
        with open(OUTPUT, 'w') as f:
            f.write(txt)
    else:
        with open('output.txt', 'w') as f:
            f.write(txt)
